Add import socket module to restraint.py