Gridworld Part 4 + 5