Fix merge, add visualisations