Create assignment 3_1.py