Some bugs fixed

Исправлены некоторые методы, которые плохо работали. Еще немного надо
поработать над праивльностью логики, но в целом более-менее