Python - Return of The Function