Remove everything
Add everything for the visualisation