Updated travis yaml and test class