Made plot_tree conform to the new signature of get_lines.