Update classification.py

remove tests