Improved travis script