Fix JSX highlighting and add colors