Add a Save button to UI