commented import pygame back in