Update mito_classofy_3D.py