python user defined function return value