Fix code blocks #15

Available in beta

Author: tobiaslins

package.json

@@ -1,5 +1,5 @@
 {
-  "version": "0.7.1",
+  "version": "0.7.2",
   "license": "MIT",
   "main": "dist/index.js",
   "typings": "dist/index.d.ts",

src/components/code.tsx

@@ -9,10 +9,12 @@ const Code: React.FC<{ code: string; language: string }> = ({
   const prismLanguage =
     languages[language.toLowerCase()] || languages.javascript;
 
+  const langClass = `language-${language.toLowerCase()}`;
+
   return (
-    <pre className="notion-code">
+    <pre className={`notion-code ${langClass}`}>
       <code
-        className={`language-${language.toLowerCase()}`}
+        className={langClass}
         dangerouslySetInnerHTML={{
           __html: highlight(code, prismLanguage, language)
         }}