adding in the class