Pygame prototype