Add code highlighting