Fixed comments

Fixes #9.