Fixed commenting issue.

Thought of sending in a PR as well.

This fixes #4.