Added ability to save and load data