Improved code comments