Added some UI code.