Clean up Mongoose model code