Added some comments

Mostly just to tick the boxes in the assignment :-)