Enhance code documentation