Improved the travis script