Code changed slightly.