change code works correctly