Creating Lambda Function Python