Added commandline options to main.py