Tweaks to plotting functions in stat_script