Improve the visualisations methods and include more visualisations on labeling