tweaks to plotting functions