Updated script for Travis CI