Add TestPlan class