Поработал с кодом, хардкод типа 1, 2... заменен enum-ом