Switch from user input to sys.argv