Add arguments to mains.py