Added class assignment