Drawing graphs with Pygame (Python)