homework 4 - last Alena comments fixes