Using Lambda For Python Code