Fix running main class