Update mito_classify.py