Update test logic.
Main for extractor demo