re-add main class