adding in class changes