Finish cs265/lab8/py_tree