Update chapter01/init_data.py