ENH: plot tweak