Add support for Mongoid