Added a button to get back to main UI