Some small udp changes