Small changes to code.