Add partial support for save in vs.py