diff --git a/main.py b/main.py
index f6a3c09..a25a535 100644
--- a/main.py
+++ b/main.py
@@ -43,12 +43,8 @@ plt.show()
 plt.clf()
 
 # Alle Helligkeitswerte der Lunge seperieren
-lungBrightness = []
 lungsArray1D = lungsArray.flatten()
-
-for (ind, elem) in enumerate(lungsArray1D):
-    if elem > 0:
-        lungBrightness.append(thoraxArray1D[ind])
+lungBrightness = thoraxArray1D[lungsArray1D > 0]
 
 # Histogram aller Helligkeitswerte in der Lunge 
 plt.hist(lungBrightness, bins=100)