WIP on a game with Python. Learning pygame