Small improvements to documentation