Add comments. Fixes #7