Optimized a bit using numpy broadcasting. Explain broadcasting.