Made changes to code