coding a game in python