Added update test and fixed save test