Updated all models to use mongoose