GridWorld Part 2 code