Added solution for step 6