FIX: correct other instance of np.linalg.norm, clarify comments