поправил баг
еще немного правок