Add JS highlighting