+IDisplayController and implementation.
Updating unit test to use IDisplayController.