More tweaks

Now it makes an error plot too, which is nice.