Initial commit of python version of problem_003