Simple video game to learn Python (PyGame)