Code for my class