Improve parameter names in main.py.