Добавлен assert
Упростил написание условий