updates to UI and add button