Photo organizer in python