Adding UDP functionality