Simple Python game engine (using PyGame)