Update JSX code