Removed the pygame import