PyGame Ant Colony game