making suggested changes to code