Update warmup1.py