Update main class reference