added support for button click to viz