Add option to set port number