Changes for task 4