Removed unused pygame import