Update dihedral.py