Add class for custom field