Add support for command line arguments