Updated to view helper

@mchesterkadwell - quick refactor, what do you think. Can you see what I have done?