Create assignment 5_2.py