Update comments

The comments now make more sense