Change so we only have to specify port for TCP and UDP