Added python script to run the whole thing