Update pygame.py

fixed if/else