added save button in UI