Modifying for Travis CI