Added command line args to python script