Add test for sample output