Add tests for list_display