Tidied up comments

Cleaned up some of the comments