Missed out on assertions for testDisplay()