Add comments

Add simple comments, otherwise the code feels rather empty.