added tree visualisation