Fix: code highlighting