Пофикшены баги, немного рефакторинга