Update 01_scalar.py