В преамбулу добавляем пакет для подсветки кода:
\usepackage{listings} % для подсветки кода
Подсвечивать так: