js-demo: various improvements in ast visualuzation