Fixing a few code highlighting issues