Добавил делегаты, немного изменил дизайн