Got basic Gridworld working