Adding logic for a test harness