Some changes, working with mongoose