refactoring

Укоротил If'ы