Create L3 PROBLEM 5B.py