Updated python test script. Connection still not closing after script ends, need to fix that.