Add code highlighting feature