Corrected main class in run for build