Cleaned extra crap from classify.py