добавил немного стейт машины