made changes to the file to complete the code