Python game using pygame api