Update blech_basics.py