redo set 1 challenge 1 with better util for python 3