corrected the class def