minor changes in the files
change in the logic