[Nexus]

рони, 54-я строка - крутое решение, я до такого не додумался)
Что element.closest может еще и "себя" вернуть не знал.

Вы избавились от абсолютного позиционирования блоков .square, считаете, что ТС оно может быть вовсе не нужно?