малко  рефакториране на кода