Home work

Этот код очень похож на верхнюю часть файла - попробуй убрать дублирование