Extracted test running into a seperate class