Optimizacion de clases

Se agrego import a la clase para poder usar pygame en la misma