updates to code