Added solution for step 1