update so that we can save allocation individually