def and return function in python