Updates to code