BUG: @dmitchell412 have to manually index the 2D array