Write Program Using Numpy Broadcasting Rules