Add python testing script