Changing the code

edit a few lines