Basic particle simulator with Pygame