Adding test for display