Add ability to set the port