Playing with making a simple game in python