further updates to code