Minor: add command line argument