simple game code in python