Fixed comments

Fixes #2