changes to the UI code