Add "save changes" functionality