Just a tad refactored. Just to swap out the mocked display with the real thing.