Using map() in Python #shorts