Added command-line arguments to tree_test