Add task 4 for week6