Update main class