Correct the class code