edit code for ui changes