Add code highlighting

Closes #45