Update to allow for visualising any data