Added buttons to the UI and added button functionality