Add classes for mongoid based tests