Add saving storage to NBT on block break. A bit dodgy atm but it works