Move python stuff to seperate file