Илья, думаю, необходимо заранее создать все необходимые стадии и разместить по 1-му элементу, с типами которых будем работать на каждой стадии. Далее можно сделать по аналогии.
Но это все по-моему костыль, потому что необходимо вносить изменения в самом скрипте, с помощью питона, думаю, можно было бы сделать получше. А самое простое решение, кажется, выделить все элементы второго этажа и присвоить им нужную стадию)