Create pygame.py

Exercise for learning python