Add in class assignment