Using Lambda Function Python