got kokkos version working for 5x5