Create L5 PROBLEM 3.py