Save fnction now working