Improve travis test script