Update calc_wiggle_correlation.py