updated highlighted code