Updates for new run script