GridWorld Part 3 Code