Update main.c

Updating to use TCP