Add UI code