Worked with GroupGP to make the array shapes fit