Нужно немного переделать рабочий код.
Сейчас при клике по любому квадрату(.box) к нему добавляется класс .bg с соответствующим номером, а при повторном клике класс .bg снова удаляется.

Как...