Немного поправил интерфейс