Tidy up comments slightly