completed warmup.py