minor changes to program