Refactor dataStore