add second part of task 4