Change from socket import to just import socket