Move data store init to MemoryStore, make it possible to use other data stores