Add in updateItemStack method