redo set 1 challenge 3 for python3