When running the code, the following error message gets displayed: `main.py: error: argument --output_fun: expected one argument`