update main.py

add the author