Now working with mongo instead of in memory