Add test for Save function