Add button code to UI