Session 1 HW

This is the task 4 assignment from session 1