Tidy up comments for functions