Add mongoid support