Remove warning for Rails 4.1

If we’re in Rails 4.1, use `options[:class]`. Else, define a `setup` method that Rails will call.