minor changes before Assignment #2 due.