python return a function with arguments