new code for class