Fixes main class