Add custom validations for models