Add source highlighting