Updated the class file