Typo fix
Simplify main.py