add changes to code