Improve Travis script