Add comments,and little improvment