Add support for mongoid