Изменил немного код

Попробовал изменить код и опубликовать