Update .travis.yml and fix imports