Sample tree visualization code added.