Initial commit of python version of prob 3