predict.py now gathers badge counts