Updated UDP code