Modify lwta.py.