Update drop_duplicates.py