Fixing imports to start the main class