Re-add main class