Небольшие правки и рефакторинг.