Added simple tests for save and load functionality.