Update lockin.py