wk1.py