Working on the visualisation tool with Density Lab