добавил один класс. Надеюсь там все правильно