grid challenge hackerrank solution in python