Update main.py

add sys