add comments, clean up code.