Task 2.1 coded