Added the same code for square 2-6