Update heads_shoulders.py