Initial gridworld task code