Add command-line argument to specify port.