Update to code functionality