UI tweaks on graph