add comment & bug fix