Adding initial python script to repo