Update main.py

refineSTs added