Python Write Function Return Value