Updating TravisCI so unit tests pass