tweak the Python vis notebook