Update main-code-vaish.py