Update 0001_initial.py

2