work on task 2 and 4