diff --git a/sz.sh b/sz.sh
index c1e4ba4831..433baeb57f 100755
--- a/sz.sh
+++ b/sz.sh
@@ -1,3 +1,5 @@
 #!/bin/bash
 # switched to cloc due to https://github.com/boyter/scc/issues/379
-cloc --by-file tinygrad/*
+cloc --by-file tinygrad/* | grep "tinygrad"
+# also some sloccount for a dir summary
+sloccount tinygrad | grep "python"
diff --git a/tinygrad/image.py b/tinygrad/nn/image.py
similarity index 99%
rename from tinygrad/image.py
rename to tinygrad/nn/image.py
index 34c3b8653e..88d7c1404a 100644
--- a/tinygrad/image.py
+++ b/tinygrad/nn/image.py
@@ -24,7 +24,6 @@ def image_dot_decorator(normal_dot):
 
 def image_conv2d_decorator(normal_conv):
   if IMAGE == 0: return normal_conv
-
   def image_conv2d(self, weight, bias=None, groups=1, stride=1, dilation=1, padding=0):
     (bs,_,iy,ix), (cout,cin,H,W) = self.shape, weight.shape
     rcout = cout//groups
diff --git a/tinygrad/tensor.py b/tinygrad/tensor.py
index 3b27516cef..b79e0a9537 100644
--- a/tinygrad/tensor.py
+++ b/tinygrad/tensor.py
@@ -5,7 +5,7 @@ import numpy as np
 from typing import List, Tuple, Callable, Optional, ClassVar, Type, Union, Sequence
 from tinygrad.helpers import prod, argfix, make_pair, getenv, DEBUG, flatten, DType, dtypes, LazyNumpyArray
 from tinygrad.lazy import Device, LazyBuffer
-from tinygrad.image import image_conv2d_decorator, image_dot_decorator
+from tinygrad.nn.image import image_conv2d_decorator, image_dot_decorator
 
 # An instantiation of the Function is the Context
 class Function: