Added tree visualisation