Update main class return