Adds support for classes

index.js

@@ -85,6 +85,8 @@ async function main() {
         for (let element of elements) {
 
             let box = await element.boundingBox();
+            let classNameAttr = await element.getProperty('className');
+            let className = await classNameAttr.jsonValue();
 
             // Scale the bounding box
             box.x /= size.width;
@@ -95,6 +97,10 @@ async function main() {
             // Give the selector as type
             box.type = selector;
 
+            if (className !== "") {
+                box.class = className;
+            }
+
             info.push(box);
         }