Updated Gridworld Project code