Fix up some of the maths docs