redo set 1 challenge 2 using python3