Updated code highlight