Add Matt's visualisation stuff