adding solution to step 5