Fixed commenting issue. Issue #4.

Thought of sending in a PR as well.