Adding UI code