Adding a python script that will help later
Adding stuff to ensure we only sample leaves