Improve model mongoDB