Update comments. fixes #1