Working on PyGame Visualization