Update vq.py