Added save button functionality