Python Return Function Using Lambda Functions