Updated Unit Tests to Capture Output Correctly.