Fixed an issue with the 2D array