Final commits for python testing functionality