Minor tweaks to get the program to run.