Minor updates to code commenting