fixing issue #3 - step 4