diff --git a/src/main/java/net/pietru/omni_power/ui/CustomScreen.java b/src/main/java/net/pietru/omni_power/ui/CustomScreen.java
index 391091f..d7e174f 100644
--- a/src/main/java/net/pietru/omni_power/ui/CustomScreen.java
+++ b/src/main/java/net/pietru/omni_power/ui/CustomScreen.java
@@ -69,16 +69,14 @@ public class CustomScreen extends BaseItemScreen {
     }
 
     static void build_ui(JsonValue layout,Table table, CustomScreen screen){
-        Iterator<JsonValue> iter = layout.iterator();
-        while (iter.hasNext()){
-            JsonValue data = iter.next();
-            String type = data.getString("type","");
+        for (JsonValue data : layout) {
+            String type = data.getString("type", "");
             if (type.isEmpty())
                 continue;
             if (!components.containsKey(type.toLowerCase()))
                 continue;
-            TriConsumer<Table, CustomScreen,JsonValue> component = components.get(type.toLowerCase());
-            component.accept(table,screen,data);
+            TriConsumer<Table, CustomScreen, JsonValue> component = components.get(type.toLowerCase());
+            component.accept(table, screen, data);
         }
     }