Add test for init universe size 2x2
Add test for init universe size 3x3