improve visualisation

improve visualisation