Updated weak_scaling script to take arguments.