Add code coloring