Update kriging.py