Updated the class name in main function.