neatened up ui a little