Add UDP server/client code