Improve the traivs test script