Slight improvements to doc