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