Add python version of program