Tree making complete in gen.py; need to prune