Basic code highlighting functionality