Add working tests for output