Add support for mongo query