First game in pygame and python