Corrected the name of the class