[TASK] Improve wording in code comment