Adding code highlighting