diff --git a/jailbreak_llama.py b/jailbreak_llama.py
index 2b588d3..58b1e0d 100644
--- a/jailbreak_llama.py
+++ b/jailbreak_llama.py
@@ -91,8 +91,8 @@ def main():
     benign_embed_generated_loc1 =[]  
     benign_embed_generated_loc2 =[]
     adverse_embed_generated=[]      #layer-wise representations
-    adverse_embed_generated_loc1=[]  #mlp-wise representations
-    adverse_embed_generated_loc2=[]  #head-wise representations
+    adverse_embed_generated_loc1=[]  #head-wise representations
+    adverse_embed_generated_loc2=[]  #mlp-wise representations
     
 
     
diff --git a/check_npy_shapes.py b/visualization.py
similarity index 100%
rename from check_npy_shapes.py
rename to visualization.py