Minor improvements to umtrx_lms.py.