Modify UI.hs