No more loops.

Training and testing files both fixed.