Modify demo to be able to loop over multiple tests.