GridWorld Part 2 done