Adding the UI code