added class assignment
Updated