Create homework2.py