Bugfix (зависало)
Some refactoring