improvements to the run script