Program Using Lambda Function Python