Added comments

Forgot to add comments to some of the code before committing. Oops!