Small update to allow style highlighting