Добавлен необходимый функционал