Added test for extract_output