Change switch statement to show proper template