add support for command line argument