Python Write Function Output Python