Tree demo fully working now