[Doc] Improved code comments