Added mongoid support