update save() function
update save() function 2