fix bug and add comment