Create work2.py