Update lm.py