adding solution for step 1
adding solution for step 1 (2)