more work on challenge 5