Having fun with pygame / python