Update sqlviz.py