TCP and UDP server, add some code