Update script for training size exp