Polish up assignment ui.