Added model classes for MongoDB