minor updates to code