python assign function to variable with arguments