Adjusted again to fix it so it trains against COUNT