Improve travis script