Add test for result display