Update lr.py