modify main.py

add print to main.py