Remove the UI code