Improve code comment