Changes to code