Added tests for UDP, but have memory leak