Add class mappings