Adds rollback testing