Прорефакторил код и добавил главную