Add test_display