Added comments

Closes #3