Updated problem description. Corrected Grid.set_agent