Updated DataStoreTest