Update for Travis integration