Bug fixed with diagonal 1's.