Added a few comments.
Minor cleanup.
updated comments