Distortion vector update

Now also plotting arrows (I hope)