Small corrections to code.