Tweaks to ploting