Minor update to how code is displayed