Added argument handling to the python test