Improvements to socket_test.sh script