added part1work2.py