First try at building a game in python