Update initialize.py