Add support for protected_attributes