Added provided gridworld code