Update SHM.py

Added fancy matplotlib vector field plots.