Add the comment, and removed some unnecessary code