Add command line argument to set port