Update moments.py