changes to the code