Small changes to the running script