Created a function to import pygame for those who don't want to install pygame