Updated MainTest: Change DisplayController to ViewController