My first game in python