Update and rename test2.py to socket.py