Updated for travis ci