change the code

Task 2