Update 'Arbitrary task' UI to match recent generic task UI update