Added solution for step 5