update comment and minor change