improve comments
add more comments