modify main.py for logger