updates to code
