Add support for setting command line arguments