Объединил 2 функции в 1. Поправил мелкие баги