Updated the main class.