Refactored models
Switched to mongoose for MongoDB document models