Code for the class