add comment and minor fix