python code from Wednesday warmup