Create L3 PROBLEM 5A.py