adding in class assignment for class 4