small changes to main run file