Add ability to save and load test data