Modified NBT logic to be a wrapper class