Update to accept command line argument for port