Still trying to get unit tests for output functions