Adding comments

Fixes #1, resolves by adding the necessary comments