Improved comments for code documentation