Add class, add weight working