Added JS for visualisation