Я учёл комментарий и попытался немножко подправить

Мне было лень сохранять в отдельный файл, поэтому держите тупо скрин