Подскажите можно ли сделать код компактнее? я пришел к такому решению.