Update calculate_rmsd.py