fix argument passing to main function

[kukas]

I wanted to call your script from my bash script using
python main.py 0.2 input output
but arguments didn't work. This commit fixes that issue.