Minor update to the ui script.