adding more args in the python script