Update to render code