Update sort_photo.py