add code for class