Improve test.sh