Update norm.py