Add support for specifying host from command line