modified main.py for some format problem