SwitchTest class edited with Switch structure