Added in yaml file for travis CI