fix task 4