Add port to the ssh.connect() in restraint.py