edited some piece of code