Adding udp support, along with new better awesome tests for tcp.