adding command line args to main shim.py file