adding a size option to run
looking to calculate variance of energy prediction