Improve comment

Improve comments