Amended to work with Travis CI

Have amended the plugin to work with Travis CI