Improved code documentation