redo set 1 challenge 5 in python3