Adding support to the mongo driver