Minor tweak to plots