Lesson 4 updated

Правки по 4 заданию
см. коммент про добавление тестового кода