Можно было бы добавить возможность выбирать кнопки состояния, а то есть только состояние Normal.
Или добавить менеджер состояний или массив.