Update .travis.yml, fix imports and tests