Simple python game code for beginners