Adding necessary visualization code for the demo.