Update main.py

Support for SAS