added gridworld part 4