Fixed training run errors. Still isn't properly generating probability table...