Add UDP socket support