Minor changes to program.