diff --git a/gen_mem.pyc b/gen_mem.pyc
index 3f24db5..17904e2 100644
Binary files a/gen_mem.pyc and b/gen_mem.pyc differ
diff --git a/gen_nn.py b/gen_nn.py
index e2f1b7c..ad90a7c 100644
--- a/gen_nn.py
+++ b/gen_nn.py
@@ -3,8 +3,15 @@
 
 args = None
 
+def gen_one_hot():
+    one_hot = np.zeros((8, 8))
+    for i in xrange(8):
+        one_hot[i, i] = 1
+    return one_hot
+
 def gen_nn(path, shape, lower=None, upper=None):
     nn = np.random.randint(lower, upper, shape, dtype=np.int8)
+    nn = gen_one_hot()
     print(nn)
     np.save(path, nn)
 
diff --git a/one_hot_label.npy b/one_hot_label.npy
new file mode 100644
index 0000000..8fbb7f3
Binary files /dev/null and b/one_hot_label.npy differ
diff --git a/one_hot_train.npy b/one_hot_train.npy
new file mode 100644
index 0000000..8fbb7f3
Binary files /dev/null and b/one_hot_train.npy differ
diff --git a/reference b/reference
index 11cc361..69abd9b 100644
--- a/reference
+++ b/reference
@@ -11,27 +11,27 @@ p3
 tp4
 Rp5
 (I1
-(I4
+(I8
 I8
 tp6
 cnumpy
 dtype
 p7
-(S'i1'
+(S'f8'
 p8
 I0
 I1
 tp9
 Rp10
 (I3
-S'|'
+S'<'
 p11
 NNNI-1
 I-1
 I0
 tp12
 bI00
-S'\xc9\x88jg\xeb\x8f\x142i\xe0(yP\x0e\x99(\xe3f\x99Yl\xc8\xf6\xc85z\xba\xee=\x80\xb5\xbf'
+S'\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?'
 p13
 tp14
 bg0
@@ -42,12 +42,12 @@ g3
 tp16
 Rp17
 (I1
-(I4
+(I8
 I8
 tp18
 g10
 I00
-S'\x00\x00\x00\x01\x00\x00\x01\x01\x01\x00\x01\x00\x00\x01\x01\x01\x00\x00\x00\x01\x01\x00\x01\x00\x01\x01\x01\x00\x01\x01\x00\x01'
+S'\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?'
 p19
 tp20
 bg0
@@ -58,42 +58,29 @@ g3
 tp22
 Rp23
 (I1
-(I4
+(I8
 I8
 tp24
-g7
-(S'f8'
+g10
+I00
+S'\xfcC\xa4\xe4\xe0\xfe\xef?Q\x13\xa5N>9*?\x98\xbd-\xa5\x138*?/\xe3b\n\x9e8*?s\xbb`\xb6Y8*?\x0bu\xbfg@8*?-\xf6O\x9bk8*?\xcd\xe2\x18\x11\xb08*?p\xeaR\xee\xc48*?P\x18\x00\xe8\xe0\xfe\xef?\xfb\xc2\xc6\xe8\xcd8*?\x84\xcd\xce\xe619*?s\x87\xc2\xb7s8*?\x0bi\xca,t9*?\xeegn#!8*?\xde&\x8b\xa3,9*?\x91\x7f\n(\xcd8*?F0\xcd\xbe\x039*?b\x16\xbb\xdd\xe0\xfe\xef?M\xcf4\xe5p8*?\x9eV\x8c!Z9*?\xd9\x9b\x0b\xb4\x919*?\xe6\xda\xf9\xb3\xa28*?\xfa\x03\xcfl49*?\xe6\xe8\xbb\x99s9*?gf\xb0\xa3y9*?g\x15\xe6\xf3<8*?\xcb\xd5g\xdb\xe0\xfe\xef?[\x16:6d8*?\xc8F\xec.t9*?:\xca\xf7<C8*?9\x99\xb9\xab\xce8*?\xd1E\x02Y\x8e9*?\x85_\xc2#\xfa8*?\xb4\xe4WD49*?\xf5\x96K\x94\xa38*?\xbb\xebc\xec\xe0\xfe\xef?\xf1\x9d\x84\x9ee9*?3\xc4E\xa9\x1c8*?\x82\xbd\xf3\xb7H9*?W\xc97V\x989*?.s\xff\x07H9*?\x83\xa0\x1b\xc3\x948*?\xd6\x12\x826V9*?\x18o\x10\x88E8*?L\xf9x\xe5\xe0\xfe\xef?\xcc\x80\xa8@~9*?(\'w[\x9a8*?\xc7\xb0\xa3\xe1\x978*?\xb9\xee"\x04R8*?\xd3\xb2x1\x1d8*?\x02r\xb6\xa0/9*?q\xb87\xe5v8*?;\xed-c\x8e8*?L\xe5\xab\xe6\xe0\xfe\xef?\x1b\x19=\xc9-8*?\xb9\xddL\x80\t9*?\xf6\xea,\xb3Y8*?\xce\xfb\xa9!\x0f9*?P\x042\x0f79*?\x18\xf2\xe7\x1cE8*?Q\x17J\xdc\xcb8*?B]l)59*?j\xd9\x90\xe4\xe0\xfe\xef?'
 p25
-I0
-I1
 tp26
-Rp27
-(I3
-S'<'
-p28
-NNNI-1
-I-1
-I0
-tp29
-bI00
-S'\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\xf0?\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\x00\xf0?'
-p30
-tp31
 bg0
 (g1
 (I0
-tp32
+tp27
 g3
-tp33
-Rp34
+tp28
+Rp29
 (I1
 (I8
 I8
-tp35
-g27
+tp30
+g10
 I00
-S"\x86\x162\x1fn\x9aa@@\xb0\xd8\x83)q\xfd\xbf^\x1bStE\x8fa@\xb5\xc0[I\x8eve\xc0\xbf\xc9.\xc1\x97\xd0#\xc0\xde\x90\xbbP\n\x90a@\x99Y\xd7}\xd4\xc3W\xc0\x16!jo\x03\x86G@\xd6\r\xf3+%GM@;\xf0\xbd\x82W\xd7Q@\xe8={\xc3\x1b^M@d\xda\xcb\xd17\x1c7\xc0\xa5\xec\xef\xe4\x8c\x8ae@\x16\xa8\x16\x0c\xc2\x84M@\x82\xab\xd6\x84\xb4\x94G\xc0`<\xb9\x1b\xb8\xca'\xc0\x04f=z\x9f\xc60\xc0L\x9f\x15pvb7\xc0\x94\x13\x1e-\x92$1\xc0\t?\x95\x05\xb6\xde;\xc0\xb9\xdf\xfd\xad\x05\x9ea\xc0p\x13q\xf9\x9390\xc0 !d\xae\xe6\xcc\xf7?\x00\xc5'E\xb7~M@\x96\x889\xae(\xffU\xc0s\xee\x81\x13\x16Te\xc0j\xcb\x90\x87I3V\xc0\xec{\xe3 ZOP@!\xf7\x07\x0c\xac\x92e\xc0\x10\x02#\xf4\x0b\xffU\xc0\x85c\x8b\xa1<8c@\x9c\xe4\xa8\xfd\xfa$?\xc0\xd8k\xc1\x0cUl\x1f@p2\xc8g\xf9\xf9R\xc0\xb0\\:MN\xac\x1e@\xc8TY\xe0A\x0e\x15\xc0\xcb\xa9\xa6\x88\xf9\xe1'\xc00\x1c=+\x1c\xee\x1e@\xe7\xed\xe0J<vJ@\xf0\xc4\xe8A3\xb4;\xc0\xe5Y\x99\xacj\xe2P@D\x83#-\x08#X\xc0H\xe7\xf6\xa1\x89\xaaP@$\xaf\xf4\xdc\xb4^<\xc0\xc0\x05\x8e8\x87\xf3 @\xd1e0v\xcb\xbfP@TV\xa6q\x12\x042\xc0\x8c\x7fA\x01\x80\xe8a\xc0\x02\xbc\xf3\xb9u\x01X\xc0x\xc7,Ff\xf5\x1e@/\xc3\xabb\xd0\x18X\xc0\x1c\xaa\xe41\xd5Y`@\x97@\x83\xb9\x99c7@\xd0\xf5\xc9\xeb\xe1\x02X\xc0\\\xb3\x9fn\x15\xa9K@$V\x97\xb9r\xd1L\xc0\xc80Q\xcbz(\x1e\xc0TI\x0b\xc2\x02\x94J\xc0\xc0O\x03Y\x91n\x1e\xc0\x14v\xdc\xda\x1f\x0c(\xc0r6\x93'\x80\x8cX\xc0L|\xa1\xfca\xcc\x1e\xc0\xf0\xb4\xc9\xc8\x9461@\x80\x02\x9c\xab\x16\xa5\xf5?"
-p36
-tp37
-btp38
+S'\x9a\xc2y\xa7\xda\xca!@Z4\xf2g\x92\x08!\xc0\xfa\xe1:0\xa9\x08!\xc0\xa1\x89\x83\xa1\x9e\x08!\xc0\x061\xdd\xd7\xa3\x08!\xc0\x11\xec\x16\xc6\xa5\x08!\xc0\xa6\xd8hz\xa2\x08!\xc0\xff)\x7fA\x9d\x08!\xc0*\x86\x10\xaa\x9b\x08!\xc0|n\xe8\xa4\xe0\xca!@\x91\x17\xbd\xfa\x9a\x08!\xc0\xca\xeb.Z\x93\x08!\xc0\xa5\x89\x02\xdc\xa1\x08!\xc0,C\x1fL\x8e\x08!\xc0>~\xb6(\xa8\x08!\xc0\xb1L\xf2\xc0\x93\x08!\xc0\xd9\xc1p\t\x9b\x08!\xc06\x1dw\xdf\x96\x08!\xc0\xa6\xa0\x84T\xce\xca!@D\x0b!\x13\xa2\x08!\xc0\x9d\r\xa7H\x90\x08!\xc0F\x88\x90\x0b\x8c\x08!\xc0%\xf3vF\x9e\x08!\xc0\xab\xcd\xe8(\x93\x08!\xc0\xdc\xa6VW\x8e\x08!\xc0\x99\xc9l\xe1\x8d\x08!\xc0H\xd5\x83\t\xa6\x08!\xc0\xc4\xc6\x13/\xca\xca!@\xea\x7f\xd2\n\xa3\x08!\xc0\xbc\xa0\xf5K\x8e\x08!\xc0\xa0\xf1\xc3\x8e\xa5\x08!\xc0\xe15\xde\xeb\x9a\x08!\xc0\r\x94\x15M\x8c\x08!\xc0\x9d\xfe\x08\x9b\x97\x08!\xc0\xa6\xf2\xfe+\x93\x08!\xc0"kZ5\x9e\x08!\xc0P\xccBy\xe8\xca!@\x8f\xc7Uh\x8f\x08!\xc0V\x87&\x80\xa8\x08!\xc0?\xa6\xa6\x9c\x91\x08!\xc0\x84\x19\x0b\x8a\x8b\x08!\xc0\x80_\x12\xaa\x91\x08!\xc0\xde~\xb4V\x9f\x08!\xc0tw(\x95\x90\x08!\xc0\x86n\xfaa\xa5\x08!\xc0U\xec\xd0"\xdc\xca!@\x18\x9aY\x87\x8d\x08!\xc0\xb2\x12q\xe9\x9e\x08!\xc0\xaa\x91\xca\x19\x9f\x08!\xc0\xc4\x02*n\xa4\x08!\xc0\xe7\x9c\xc2u\xa8\x08!\xc0\xb33\x95\x86\x93\x08!\xc0\\\xd2\xf4\x9d\xa1\x08!\xc0\xacL1\xd3\x9f\x08!\xc0\x88\xe5.F\xde\xca!@\xbf\xaf\xb51\xa7\x08!\xc0\xab"\x12o\x96\x08!\xc0"\xbb\x1b\xd8\xa3\x08!\xc0\xda\xbf \x01\x96\x08!\xc0\xdcmx\xf5\x92\x08!\xc0e$\'j\xa5\x08!\xc0"\x15\xbf"\x9b\x08!\xc0\x04\xd4\x85\x1a\x93\x08!\xc0\x12d\xd9\x84\xda\xca!@'
+p31
+tp32
+btp33
 .
\ No newline at end of file
diff --git a/weight_dram.npy b/weight_dram.npy
index 07d699f..f628de9 100644
Binary files a/weight_dram.npy and b/weight_dram.npy differ