improve lambda lifting