Adding display back in