added in class assignment part