did suggested changes to my code