Switch from tsung class to driver