Tweaking the run script to work with your code.