Update pls.py