made changes to the code