Fix class add and remove bug

Some of the field names didn't match up so classes weren't being added or removed.