Add models support for MongoDB