Switch DB to Mongo and fix models