Added Task 2 code