Create assignment3.1.py