Add option to input token through command line