Added python script for program testing