Added solution for step 4