more idiomatic challenge 5
tried to use a more ellaborate system
get challenge 6 to work