Added setup & teardown using custom class
Updated travis config