Remove unnecessary main class definition