Add the DataStore test logic