classify.py bug fixes