Update InMemoryDataStore.scala