Adding save button to the UI