Updated visualisation section in line with jonnys example