Improve code comments