Improve comment module