Fix few mistakes for 10th homework