Add class assignment