Update Task 1 the code