Update material budget estimation codes

Material Budget estimation