Slight improvements to code documentation