Restructured so that it's just to swap out the mock render with the actual device