chore: tidy up comments