Here you can just use ``np.ones()`` to create a 2D array directly (more efficient)