simplified by adding np.zeros