minor corrections in main.py