Cleaned up division by using np.divide)