update homework 2

Signed-off-by: Andrew Tselischev <andrewtselischev@gmail.com>