Add initial load API
Add a test for datastore.load