adding in class to do