Update screenlock.py