Improve comment