Update calculate_rmsd.py

Updated while loop