diff --git a/.vscode/settings.json b/.vscode/settings.json
new file mode 100644
index 0000000..29819fc
--- /dev/null
+++ b/.vscode/settings.json
@@ -0,0 +1,20 @@
+{
+    "MicroPython.executeButton": [
+        {
+            "text": "▶",
+            "tooltip": "Run",
+            "alignment": "left",
+            "command": "extension.executeFile",
+            "priority": 3.5
+        }
+    ],
+    "MicroPython.syncButton": [
+        {
+            "text": "$(sync)",
+            "tooltip": "sync",
+            "alignment": "left",
+            "command": "extension.execute",
+            "priority": 4
+        }
+    ]
+}
\ No newline at end of file
diff --git a/src/interface.jl b/src/interface.jl
index 3fb8b7d..ed17d18 100644
--- a/src/interface.jl
+++ b/src/interface.jl
@@ -37,11 +37,11 @@ function Base.getindex(container::filecontainer, idx::Int)
 
     _img = Images.load(path)
     # _img = apply(tfm, Image(_img))  # augment an image
-    img = collect(channelview(float32.(RGB.(_img))))
+    _img = channelview(float32.(RGB.(_img)))
     # img = (img .- mean(img)) ./ std(img)      # normalize the image
-    img = permutedims(img, (3, 2, 1))
-    y = container.classIdOrder[Symbol(classId)]
-    return img, y
+    img = permutedims(_img, (3, 2, 1))
+    # y = container.classIdOrder[Symbol(classId)]
+    return img
 end
 
 
@@ -104,6 +104,7 @@ end
 
 
 
+
 
 
 end # module interface
\ No newline at end of file