Python Return Function Using Lambda