Update get_models_from_pdb.py