Updating main class name