add test and sample_output