Math Operations on Arrays using Numpy in Python