Update trace.py

Copied normalization code from EIT branch.