Немного стабилизировал код