Updated for travis-ci