changed class assignment back