Create warmup.py