fixed my class