Okay, now I corrected the class