Tweaks to plots and extra docs