python, warmup 2