Minor updates to code.