updates to code functionality