Update test.py

Updated the way to grab arguments from the command line. 
Using : sys.argv array is much way better than a parse.