Modify more code and add simple protocol on top of TCP to transfer data