Ревью и правки.

Поревьювил, добавил комментарии и немножко дописал код в нескольких местах.