Немного подравлен хардкод с swith