Refactored gem, added config for default_policy_class