Python 3 Write Function