Update dipole.py