Code works on a 3x3