Update loading.py