diff --git a/src/main/java/com/hbm/tileentity/machine/TileEntityMachineRTG.java b/src/main/java/com/hbm/tileentity/machine/TileEntityMachineRTG.java
index 2980ee948..1314bfd00 100644
--- a/src/main/java/com/hbm/tileentity/machine/TileEntityMachineRTG.java
+++ b/src/main/java/com/hbm/tileentity/machine/TileEntityMachineRTG.java
@@ -3,6 +3,7 @@ package com.hbm.tileentity.machine;
 import java.util.ArrayList;
 import java.util.List;
 import com.hbm.items.ModItems;
+import com.hbm.items.machine.ItemRTGPellet;
 import com.hbm.lib.Library;
 import com.hbm.packet.AuxElectricityPacket;
 import com.hbm.packet.PacketDispatcher;
@@ -215,22 +216,14 @@ public class TileEntityMachineRTG extends TileEntity implements ISidedInventory,
 			heat = 0;
 			
 			for(int i = 0; i < slots.length; i++) {
-				if(slots[i] != null) {
-					if(slots[i].getItem() == ModItems.pellet_rtg)
-						heat += 5;
-					if(slots[i].getItem() == ModItems.pellet_rtg_weak)
-						heat += 3;
-					if(slots[i].getItem() == ModItems.pellet_rtg_polonium)
-						heat += 25;
-					if(slots[i].getItem() == ModItems.pellet_rtg_americium)
-						heat += 50;
+				
+				if(slots[i] != null && slots[i].getItem() instanceof ItemRTGPellet) {
+					
+					heat += ((ItemRTGPellet)slots[i].getItem()).getHeat();
 					
 					if(slots[i].getItem() == ModItems.pellet_rtg_gold) {
-						
 						if(worldObj.rand.nextInt(60*60*20) == 0)
 							slots[i] = null;
-						else
-							heat += 150;
 					}
 				}
 			}
diff --git a/src/main/java/com/hbm/tileentity/machine/rbmk/TileEntityRBMKBase.java b/src/main/java/com/hbm/tileentity/machine/rbmk/TileEntityRBMKBase.java
index 812d0c344..152cb65cc 100644
--- a/src/main/java/com/hbm/tileentity/machine/rbmk/TileEntityRBMKBase.java
+++ b/src/main/java/com/hbm/tileentity/machine/rbmk/TileEntityRBMKBase.java
@@ -47,6 +47,15 @@ import net.minecraftforge.common.util.ForgeDirection;
  */
 public abstract class TileEntityRBMKBase extends TileEntity implements INBTPacketReceiver {
 	
+	/*
+	 * ██████╗ ██████╗ ███╗   ███╗██╗  ██╗
+	 * ██╔══██╗██╔══██╗████╗ ████║██║ ██╔╝
+	 * ██████╔╝██████╔╝██╔████╔██║█████╔╝ 
+	 * ██╔══██╗██╔══██╗██║╚██╔╝██║██╔═██╗ 
+	 * ██║  ██║██████╔╝██║ ╚═╝ ██║██║  ██╗
+	 * ╚═╝  ╚═╝╚═════╝ ╚═╝     ╚═╝╚═╝  ╚═╝
+	 */
+	
 	public double heat;
 	
 	public int water;