Refactor some model loading to simplify it and use mongoose better