Небольшой рефакторинг кода и тестов