Исправление заполнения состояний

Исправленно не до конца